from classifier import DocumentClassifier

def main():
    # 准备训练数据
    training_documents = [
        "这是一篇关于人工智能的文章，讨论了机器学习和深度学习的应用",
        "今天的天气很好，阳光明媚，适合出去散步",
        "最新的手机采用了先进的人工智能技术，能够实现智能语音助手功能",
        "周末去公园野餐，遇到了很多朋友，玩得很开心"
    ]
    
    training_labels = ['技术', '生活', '技术', '生活']
    
    # 准备测试数据
    test_documents = [
        "新开发的人工智能系统可以自动进行图像识别",
        "昨天和家人一起去海边度假，拍了很多照片"
    ]
    test_labels = ['技术', '生活']
    
    # 创建分类器实例
    classifier = DocumentClassifier()
    
    # 训练分类器
    print("正在训练分类器...")
    classifier.add_training_documents(training_documents, training_labels)
    
    # 评估分类器性能
    print("\n分类器评估报告:")
    print(classifier.evaluate(test_documents, test_labels))
    
    # 测试新文档分类
    new_document = "这款新手机采用了5nm制程的处理器，性能提升显著"
    predicted_category = classifier.predict(new_document)
    print(f"\n新文档预测类别: {predicted_category}")

if __name__ == '__main__':
    main()